from n_gram.ProcessText import process_text
from n_gram.NGram import generate_ngrams
import sentence_cut
if __name__ == '__main__':
    text = 'A quick brown fox jumps over the lazy dog.'

    words_list = process_text(text)
    unigrams = generate_ngrams(words_list, 1)
    bigrams = generate_ngrams(words_list, 2)
    trigrams = generate_ngrams(words_list, 3)
    fourgrams = generate_ngrams(words_list, 4)
    fivegrams = generate_ngrams(words_list, 5)

    print(trigrams)